
batch_size=2
train_time=20
img_size=1024
